package com.cb2.algorithm.leetcode;

import java.util.ArrayDeque;
import java.util.Queue;

/**
 * <a href='https://leetcode.cn/problems/sum-root-to-leaf-numbers/'>求根节点到叶节点数字之和(Sum Root to Leaf Numbers)</a>
 * <p>给你一个二叉树的根节点 root ，树中每个节点都存放有一个 0 到 9 之间的数字。</p>
 * <p>每条从根节点到叶节点的路径都代表一个数字：
 * <ul>
 *     <li>例如，从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。</li>
 * </ul></p>
 * <p>计算从根节点到叶节点生成的 所有数字之和 。</p>
 * <p>叶节点 是指没有子节点的节点。</p>
 *
 * <p>
 * <b>示例：</b>
 * <pre>
 *  示例 1：
 *      输入：root = [1,2,3]
 *              1
 *            /   \
 *           2     3
 *      输出：25
 *      解释：
 *          从根到叶子节点路径 1->2 代表数字 12
 *          从根到叶子节点路径 1->3 代表数字 13
 *          因此，数字总和 = 12 + 13 = 25
 *
 *  示例 2：
 *      输入：root = [4,9,0,5,1]
 *                    4
 *                  /   \
 *                 9     0
 *               /  \
 *              5    1
 *      输出：1026
 *      解释：
 *          从根到叶子节点路径 4->9->5 代表数字 495
 *          从根到叶子节点路径 4->9->1 代表数字 491
 *          从根到叶子节点路径 4->0 代表数字 40
 *          因此，数字总和 = 495 + 491 + 40 = 1026
 * </pre>
 * </p>
 *
 * <p>
 * <b>提示：</b>
 * <ul>
 *     <li>树中节点的数目在范围 [1, 1000] 内</li>
 *     <li>0 <= Node.val <= 9</li>
 *     <li>树的深度不超过 10</li>
 * </ul>
 * </p>
 *
 * @author c2b
 * @since 2025/2/18 16:06
 */
public class LC0129SumRootToLeafNumbers_M {
    static class Solution {
        public int sumNumbers(TreeNode root) {
            //return sumNumbersByIteration(root);
            return sumNumbersByRecursion(root);
        }

        private int sumNumbersByIteration(TreeNode root) {
            int sum = 0;
            Queue<TreeNode> helper = new ArrayDeque<>();
            helper.offer(root);
            while (!helper.isEmpty()) {
                TreeNode currNode = helper.poll();
                int currNodeVal = currNode.val;
                if (currNode.left == null && currNode.right == null) {
                    sum += currNodeVal;
                }
                if (currNode.left != null) {
                    helper.offer(currNode.left);
                    currNode.left.val = currNodeVal * 10 + currNode.left.val;
                }
                if (currNode.right != null) {
                    helper.offer(currNode.right);
                    currNode.right.val = currNodeVal * 10 + currNode.right.val;
                }
            }
            return sum;
        }

        public int sumNumbersByRecursion(TreeNode root) {
            return dfs(root, 0);
        }

        private int dfs(TreeNode currNode, int prevNodeVal) {
            if (currNode == null) {
                return 0;
            }
            prevNodeVal = prevNodeVal * 10 + currNode.val;
            if (currNode.left == null && currNode.right == null) {
                return prevNodeVal;
            }
            return dfs(currNode.left, prevNodeVal) + dfs(currNode.right, prevNodeVal);
        }
    }

    public static void main(String[] args) {
        TreeNode root1 = new TreeNode(1);
        root1.left = new TreeNode(2);
        root1.right = new TreeNode(3);

        TreeNode root2 = new TreeNode(4);
        root2.left = new TreeNode(9);
        root2.right = new TreeNode(0);
        root2.left.left = new TreeNode(5);
        root2.left.right = new TreeNode(1);

        Solution solution = new Solution();
        System.out.println(solution.sumNumbers(root1));
        System.out.println(solution.sumNumbers(root2));
    }
}
